# -*- coding: utf-8 -*-
"""
创建示例数据脚本
用于生成测试用的示例数据
"""

import os
import numpy as np
from PIL import Image
from config import Config

def create_sample_data():
    """创建示例数据"""
    # 创建数据目录
    data_dir = Config.TRAIN_DATA_DIR
    os.makedirs(data_dir, exist_ok=True)
    
    print("创建示例数据目录: {}".format(data_dir))
    
    # 创建示例图片和标签文件
    num_samples = 20  # 创建20个样本
    
    for i in range(num_samples):
        # 创建随机图片
        img_array = np.random.randint(0, 255, (224, 224, 3), dtype=np.uint8)
        img = Image.fromarray(img_array)
        img_path = os.path.join(data_dir, 'sample_image_{:03d}.jpg'.format(i))
        img.save(img_path)
        
        # 创建对应的标签文件
        label_path = os.path.join(data_dir, 'sample_image_{:03d}.txt'.format(i))
        label_id = i % Config.NUM_CLASSES  # 循环使用类别ID
        with open(label_path, 'w', encoding='utf-8') as f:
            f.write('sample_image_{:03d}.jpg, {}'.format(i, label_id))
        
        print("创建样本 {}: 图片={}, 标签={}".format(i, img_path, label_id))
    
    print("示例数据创建完成！")
    print("总共创建了 {} 个样本".format(num_samples))
    print("数据目录: {}".format(data_dir))

if __name__ == '__main__':
    create_sample_data() 